﻿import platform
from tkinter import Tk

from fanqie_downloader import *

cookie_res = config_json.get('cookie')
if get_cookie('/reader/7283710075737211404', cookie_res) == 'err':
    get_cookie('/reader/7283710075737211404')

# 主循环
while True:
    print('\n输入书的id直接下载\n输入下面的数字进入其他功能:')
    print('''1.更新小说\n2.搜索\n3.批量下载\n4.退出\n''')
    choice = input()
    if choice == '1':
        print('更新完成')
    if choice == '2':
        down_book(search_book_list())
    if choice == '3':
        urls_path = 'urls.txt'  # 定义文件名
        if not os.path.exists(urls_path):
            print(f"未找到'{urls_path}'，将为您创建一个新的文件。")
            with open(urls_path, 'w', encoding='UTF-8') as file:
                file.write("# 请输入书名/书籍id，一行一个\n")
        print(f"'{urls_path}' 已存在。请在文件中输入小说链接，一行一个。")

        # 使用默认文本编辑器打开urls.txt文件供用户编辑
        root = Tk()
        root.withdraw()  # 隐藏主窗口
        root.update()  # 更新Tkinter的事件循环，确保窗口被隐藏

        if platform.system() == "Windows":
            # Windows系统使用os.startfile
            os.startfile(urls_path)
        elif platform.system() == "Darwin":
            # macOS系统使用open命令
            os.system(f"open -a TextEdit {urls_path}")
        else:
            # 其他系统使用默认文本编辑器
            os.system(f"xdg-open {urls_path}")

        print("输入完成后请保存并关闭文件，然后按Enter键继续...")
        input()

        # 读取urls.txt文件中的链接
        with open(urls_path, 'r', encoding='UTF-8') as file:
            content = file.read()
            urls = content.replace(' ', '').split('\n')

        # 开始批量下载
        for url in urls:
            if url[0] != '#':
                print(f'开始下载【小说】: {url.split("/")[0]}')
                down_book(url.split("/")[-1])
    if choice == '4':
        break

